import pprint
import unittest
from src.Class_CVE import CveSearcher

class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.cve = CveSearcher("CVE-2021-37121", "6", "aarch64")

        # self.cve = CVE("CVE-2017-2665", "6", "x86_64")
    @unittest.skip
    def test_get_cve_response(self):
        pprint.pprint(self.cve.relation_list())

    @unittest.skip
    def test_analysis_response(self):
        print(self.cve.relation_list())


    # 测试CVE类的cvereport方法
    def test_cvereport(self):
        print(self.cve.cvereport())
if __name__ == '__main__':
    unittest.main()
